from langchain.prompts import Prompt
from langchain_core.prompts import PromptTemplate


class ChatPromptHelper:
    def __init__(self):
        pass

    # 获取某个城市是否有什么好吃的
    def get_template_for_goods(self):
        template = "请问{city}有什么好吃的"
        prompt = PromptTemplate(template=template, input_variables=["city"])
        return prompt

    # 获取旅游模版
    def get_template_for_tourism(self):
        template = """
                你是一个旅游销售员，你不仅要告诉用户正确的答案，还要用诗一样的语言来描述这个答案！
                请告诉我{country}的首都是那座城市？
                """

        prompt3 = PromptTemplate(template=template, input_variables=["country"])
        return prompt3
